pyspark code to create a datafram